'''
manage.py 只做最基本的启动工作，将 app 的创建操作移动到 info 的 __init__.py 文件中
'''

import logging
from logging.handlers import RotatingFileHandler
import redis
from flask import g
from flask import render_template
from flask_sqlalchemy import SQLAlchemy
from flask_wtf.csrf import CSRFProtect, generate_csrf
from flask_session import Session  # 利用 flask-session扩展，将 session 数据保存到 Redis 中

from config import Config,config
from flask import Flask


# 数据库
from info.utils.common import user_login_data

db = SQLAlchemy()   # 为了接受参数
redis_store = None  # 设置空值的原因为：传值方便


def setup_log(log_level):
    """配置日志"""

    # 设置日志的记录等级
    logging.basicConfig(level=log_level)  # 调试debug级
    # 创建日志记录器，指明日志保存的路径、每个日志文件的最大大小、保存的日志文件个数上限
    file_log_handler = RotatingFileHandler("logs/log", maxBytes=1024 * 1024 * 100, backupCount=10)
    # 创建日志记录的格式 日志等级 输入日志信息的文件名 行数 日志信息
    formatter = logging.Formatter('%(levelname)s %(filename)s:%(lineno)d %(message)s')
    # 为刚创建的日志记录器设置日志记录格式
    file_log_handler.setFormatter(formatter)
    # 为全局的日志工具对象（flask app使用的）添加日志记录器
    logging.getLogger().addHandler(file_log_handler)


# 添加 create_app 的工厂方法
def create_app(config_name): # config_name = development
    """通过传入不同的配置名字，初始化其对应配置的应用实例"""

    # flask实例对象的初始化  ( 这个初始化函数就是初始化的头必须在init里面)
    app = Flask(__name__)

    # 加载配置类到flask项目中
    # app.config.from_object(Config)
    app.config.from_object(config[config_name])  # config[config_name] 是键 对应字典中的值为 DevelopementConfig

    # 加载配置类到SQLAlchemy项目中
    # db = SQLAlchemy(app)
    global db
    db.init_app(app)  # 忘了早说的按固定格式记吧

    # 实例化redis操作对象
    global redis_store  # 声明局部变量为全局变量 原来为全局变量
    redis_store = redis.StrictRedis(host=Config.REDIS_HOST,  port=Config.REDIS_PORT)

    # 开启csrf防范机制
    CSRFProtect(app)  # (CSRFProtect只做验证工作，cookie中的 csrf_token 和表单中的 csrf_token 需要我们自己实现)
    # 在每次客户请求成功以后，返回cdrf_token

    @app.after_request
    def after_request(response):
        # 手动生成csrf_token
        csrf_token = generate_csrf()
        # 可以通过设置cookies的方式，把csrf_token告诉浏览器
        response.set_cookie("csrf_token", csrf_token)

        return response

    # 启用Session
    Session(app)

    # 启动日志功能
    setup_log(Config.LOG_LEVEL)

    # 注册蓝图
    from info.modules.index import index_blu
    app.register_blueprint(index_blu)

    # 注册图片验证码的蓝图
    from info.modules.passport import passport_blu
    app.register_blueprint(passport_blu)

    # 新闻详情
    from info.modules.new import news_blu
    app.register_blueprint(news_blu)

    # 个人中心
    from info.modules.profile import profile_blu
    app.register_blueprint(profile_blu)

    # 注册自定义过滤器
    from info.utils.common import do_index_class
    app.add_template_filter(do_index_class, 'index_class')

    # 注册后台管理界面的蓝图
    from info.templates.admin import admin_blu
    app.register_blueprint(admin_blu)

    # 捕获404异常， 显示404页面
    @app.errorhandler(404)  # 装饰器
    @user_login_data
    def page_not_found(_):  # 在工作中，如果某些变量或者参数表示没有意义 就是之后也用不到 可以用_来表示
        user = g.user
        return render_template('news/404.html', user=user)

    # #注册蓝图到flask中（新闻详情页的蓝图的注册）
    # from info.modules.index import index_blu
    # app.register_blueprint(index_blu)
    #
    # from info.modules.passport import passport_blu
    # app.register_blueprint(passport_blu)
    #
    # from info.modules.new import news_blu
    # app.register_blueprint(news_blu)

    return app

